package pathdb

import (
	"encoding/binary"
	"testing"

	"github.com/ethereum/go-ethereum/common"
)

const benchDatasetSize = 1024

func benchmarkDataset() ([]common.Hash, []common.Hash) {
	accounts := make([]common.Hash, benchDatasetSize)
	storages := make([]common.Hash, benchDatasetSize)
	for i := 0; i < benchDatasetSize; i++ {
		binary.BigEndian.PutUint64(accounts[i][24:], uint64(i))
		binary.BigEndian.PutUint64(storages[i][24:], uint64(i*17))
	}
	return accounts, storages
}

var (
	benchmarkSink      int
	benchmarkSinkBytes []byte
)

func BenchmarkStorageKey(b *testing.B) {
	accounts, storages := benchmarkDataset()
	mask := benchDatasetSize - 1

	b.Run("storageKey", func(b *testing.B) {
		b.ReportAllocs()
		var sum int
		b.ResetTimer()
		for i := 0; i < b.N; i++ {
			idx := i & mask
			accountHash, storageHash := accounts[idx], storages[idx]
			skey := storageKey(accountHash, storageHash)
			key := skey[:]
			sum += int(key[0])
			benchmarkSinkBytes = key
		}
		benchmarkSink = sum
	})

	b.Run("append", func(b *testing.B) {
		b.ReportAllocs()
		var sum int
		b.ResetTimer()
		for i := 0; i < b.N; i++ {
			idx := i & mask
			accountHash, storageHash := accounts[idx], storages[idx]
			key := append(accountHash[:], storageHash[:]...)
			sum += int(key[0])
			benchmarkSinkBytes = key
		}
		benchmarkSink = sum
	})
}